import numpy as np
from hmmlearn import hmm

if __name__ == '__main__':
    states = [
        'b1',
        'b2',
        'b3'
    ]

    n_states = len(states)
    obs = ["red", "green"]
    n_obs = len(obs)

    stat_prob = np.array([.2, .4, .4])
    trans_prob = np.array([
        [.5, .2, .3],
        [.3, .5, .2],
        [.2, .3, .5]
    ])

    emission_prob = np.array([
        [.5, .5],
        [.4, .6],
        [.7, .3]
    ])

    model = hmm.MultinomialHMM(n_components=n_states)
    model.startprob_ = stat_prob
    model.transmat_ = trans_prob
    model.emissionprob_ = emission_prob

    seen = np.array([[0, 1, 1]]).T

